# 写一个python程序，要求把存放在“C:\Users\Kelvin\Desktop\Python\Excel\Organization.xlsx”文件
# 中，“Sheet2”表格的“换热站”字段中的内容，按照文本相似性判断，与“Sheet1”表格中的“换热站”字段中
# 的内容进行关联，并把“Sheet1”表格中的“中心站”和“管理所”两个字段中的内容复制到“Sheet2”表格中，
# 新建“备注”字段，用来保留原“Sheet2”表格的“换热站”字段中的内容。
import pandas as pd
from fuzzywuzzy import fuzz

# 读取Sheet1和Sheet2
df1 = pd.read_excel(r'C:\Users\Kelvin\Desktop\Python\Excel\2021-2022.xlsx', sheet_name='Sheet1')
df2 = pd.read_excel(r'C:\Users\Kelvin\Desktop\Python\Excel\2021-2022.xlsx', sheet_name='Sheet2')
# 对于Sheet2中的每个换热站，找到最相似的Sheet1中的换热站
for i, row in df2.iterrows():
    max_score = -1
    max_index = -1
    for j, row2 in df1.iterrows():
        score = fuzz.ratio(str(row['换热站']), str(row2['换热站']))
        if score > max_score:
            max_score = score
            max_index = j
    # 将Sheet1中的中心站和管理所复制到Sheet2中
    df2.at[i, '中心站'] = df1.at[max_index, '中心站']
    df2.at[i, '管理所'] = df1.at[max_index, '管理所']
# 将结果写入Sheet2
df2.to_excel(r'C:\Users\Kelvin\Desktop\Python\Excel\Organization_3.xlsx', sheet_name='Sheet2', index=False)
